pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
上面是官方链接,不知道为什么我科学上网了下的还是非常慢,而且看网上有下好之后还有问题的,于是果断选择清华源!
下面是一步一步教程:
首先确保安装了anaconda,在anaconda prompt命令行窗口创建虚拟环境,我命名为 pytorch310(Python版本不要低于3.8)
conda create -n pytorch310 python==3.10
然后激活它:
conda activate pytorch310
设置清华源,加速安装:
pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple
开始指定版本安装:文章来源:https://www.toymoban.com/news/detail-496185.html
pip install torch==2.0.0+cu118 torchvision==0.15.1+cu118 torchaudio==2.0.1+cu118 -f https://download.pytorch.org/whl/torch_stable.html
应该会非常快!我的下载速度平均在40M/s,两分钟就下好了!
然后输入conda list
看一下有哪些库,这三个出来就没问题了!
不太放心,再试一下:
输入python
,然后import torch
,这个时候稍微等一下有点慢,然后torch.cuda.is_available()
,如下图所示,出来True说明可以调用cuda!gpu版本pytorch安装成功!
文章来源地址https://www.toymoban.com/news/detail-496185.html
到了这里,关于GPU版本pytorch(Cuda12.1)清华源快速安装一步一步教!小白教学~的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!