自己电脑环境:
CUDA 版本:11.6
win11系统;
1.创建虚拟环境
# 虚拟环境名称:pytorch310
# python版本:3.10
conda create -n pytorch310 python=3.10
2.进入虚拟环境
conda activate pytorch310(虚拟环境名字)
3.下载.whl文件
Pytorch官网地址:
https://pytorch.org/
发现给出的CUDA是11.7和11.8,我自己电脑是11.6版本,不能直接用给出的命令。
进入下面这个地址,找到对应CUDA版本
https://pytorch.org/get-started/previous-versions/
推荐使用pip下载方式;
pip install torch==1.13.1+cu116 torchvision==0.14.1+cu116 torchaudio==0.13.1 --extra-index-url https://download.pytorch.org/whl/cu116
复制最后的链接,并在浏览器中打开(如果打开失败,可以使用科学上网)
https://download.pytorch.org/whl/cu116
打开后如下图所示,
点击“torch”,找到对应的.whl文件
其中,cp310是指python版本是3.10,对应你的虚拟环境python版本;torch-1.13.1是上面pip命令中对应的版本;cu116是CUDA版本是11.6。
推荐把链接复制到迅雷进行下载,速度比较快。
回退到上一页面,点击“torchvision”,找到对应.whl文件
同样复制链接到迅雷中进行下载
回退到上一页面,点击“torchaudio”
同样复制链接到迅雷中进行下载。
使用cmd进入到这三个文件下载的路径
如果是win11,可以直接在该文件夹下右键选择“在终端中打开”:
输入命令(注意顺序和教程中一致):
pip install torch-1.13.1+cu116-cp310-cp310-win_amd64.whl
pip install torchvision-0.14.1+cu116-cp310-cp310-win_amd64.whl
pip install torchaudio-0.13.1+cu116-cp310-cp310-win_amd64.whl
5.查看是否安装成功
python
import torch
torch.cuda.is_available()
显示True,表示成功。
6.接下来介绍如何在Anaconda的虚拟环境中使用
进入路径:
D:\Anaconda\anaconda\Lib\site-packages
把上面对应的6个文件复制,
粘贴到下面路径中:
D:\Anaconda\anaconda\envs\pytorch310\Lib\site-packages
如图所示:
7.打开Anaconda Promot,进入虚拟环境中:
activate pytorch310
输入:
python
输入:
import torch
报错如下:
缺少module,pip安装一下
解决:
输入:
pip install typing_extensions
缺少需要的包,pip进行安装
输入:
# 使用清华源安装numpy,速度很快
pip install numpy -i https://pypi.tuna.tsinghua.edu.cn/simple
继续安装缺少的包
pip install pillow==8.0 -i https://pypi.tuna.tsinghua.edu.cn/simple
报错:
解决:
输入:
pip install pillow==
安装给出的最新版本,避免不兼容问题
pip install pillow==9.4.0 -i https://pypi.tuna.tsinghua.edu.cn/simple
还缺少requests,pip安装一下:
pip install requests -i https://pypi.tuna.tsinghua.edu.cn/simple
至此,所有缺少的包全部下载完成。
8.进行测试
输入:
python
import torch
torch.cuda.is_available()
安装成功。
补充:
(1)一开始是直接使用conda的命令直接进行下载,速度非常慢,而且最后显示时间超时,下载失败。
(2)然后改用pip命令直接下载,依然由于速度太慢,无法成功。(这两种方式虽然用了科学上网,但是依然无法成功安装)
(3)然后使用清华源安装:pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu116 -i https://pypi.tuna.tsinghua.edu.cn/simple
使用该命令确实可以安装成功,但最后测试的时候输入torch.cuda.is_available()显示False。在网上查询得到可能是因为换了国内镜像的原因。
(4)最后尝试把.whl文件下载到本地的的方式进行安装,要注意的是我在下载.whl文件的时候发现使用迅雷下载的方式要快于科学上网,推荐迅雷下载。后面在Anaconda虚拟环境中使用的时候又出现了缺少包、版本不兼容的问题,进行了解决。