GPU版本pytorch(Cuda12.1)清华源快速安装一步一步教!小白教学~ |
您所在的位置:网站首页 › cuda版本与tensflow版本 › GPU版本pytorch(Cuda12.1)清华源快速安装一步一步教!小白教学~ |
GPU版本pytorch(Cuda12.1)清华源快速安装一步一步教!
作者为新电脑r9000p装pytorch时,没有注意自己的cuda版本是12.1(小伙伴自行百度下如何查看),装了旧版的torch结果import报错!找了半天原因才发现跟cuda版本不对应,要下载Cuda 11.8这个版本才可以,如下图:
![]() ![]() 上面是官方链接,不知道为什么我科学上网了下的还是非常慢,而且看网上有下好之后还有问题的,于是果断选择清华源! 下面是一步一步教程: 首先确保安装了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开始指定版本安装: 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看一下有哪些库,这三个出来就没问题了! |
今日新闻 |
推荐新闻 |
CopyRight 2018-2019 办公设备维修网 版权所有 豫ICP备15022753号-3 |