因为要使用多个环境,在安装了anaconda、cuda11.8、对应cuda11.8版本的cudnn之后。在anaconda下创建需要的Pytorch环境,隔一段时间对操作流程有些不熟练,为避免每次都花时间搜寻,以此文记录。
首先打开anaconda Prompt
然后创建虚拟环境,two为你要创建的环境名称,Python版本号可自选,此处我选择Python3.11
conda create -n two python=3.11
然后激活进入当前环境
conda activate two
然后选择下载torch和torchvision到本地安装。之所以选择这种方式,是因为之前在pytorch官网无论是使用pip指令还是conda指令都因为网络问题等安装不成功,换源也不行。使用这种方式安装较快。且文件下载到电脑后方便后续安装其它环境使用。
torch和torchvision下载网址:
https://download.pytorch.org/whl/torch_stable.html
还是以我的cuda118为例,在该网址下寻找对应版本的.whl文件,如图。
注:cu118说明是cuda11.8版本,cp311说明适配python3.11版本。win和linux就是电脑系统类别
下载成功后,依次使用指令安装验证
pip install D:\software\pytorch\torch-2.1.0+cu118-cp311-cp311-win_amd64.whl
pip install D:\software\pytorch\torchvision-0.16.0+cu118-cp311-cp311-win_amd64.whl
python # 进入Python环境
import torch
print(torch.__version__)
import torchvision
print(torchvision.__version__)
exit()
安装torch:其中install后面的为两个whl文件的目录
安装torchvision:
出现一下界面则说明安装成功
ps:删除环境指令为conda remove -n two --all
退出环境为conda deactivate
环境安装成功后需要关联到pycharm,打开设置
配置项目解释器:
使用现有环境,选择你创建好的环境即可