前言
想要安装pytorch,至少得先安装Anaconda、python!!!
必要的不想用cpu要用gpu的还需要cuda11.0+cudnn11.0!!!
一、安装python3.9
二、安装Anaconda3
三、安装cuda11.0+cudnn8.0
四、安装pytorch
在https://download.pytorch.org/whl/torch_stable.html里找对应的torch:
找这个:
下载到D盘。
打开Anaconda Prompt
创建Pytorch虚拟环境。(记得创建环境的默认是C盘,要更改一下到D盘因为太占位置。更改方法在此)
输入以下指令:conda create -n pytorch python=3.9。
会提问我们是否安装一些新的包,我们输入y,然后回车,等待一会就能安装成功~
安装成功之后我们激活pytorch环境
输入指令:conda activate pytorch,激活之后就不用再管Anaconda了
将Pycharm2019.3的python解释器换成激活的torch环境(因为是老版本的Pycharm,将我们的3.9自动降为3.8了)
附上 带有Anaconda插件的Pycharm2019.3
链接:https://pan.baidu.com/s/1wzS-JtyXgTTGHOLsclOTSg?pwd=nhg8 提取码:nhg8 --来自百度网盘超级会员V4的分享
1、打开PyCharm,进入刚刚创建的工程里,打开Terminal窗口:
2、在PyCharm的Terminal窗口中键入cd 命令,切换目录到PyTorch安装包所在的地址,然后分别执行pip install命令安装两个文件,先安装torch,再安装torchvision,期间会一起下载所需要的依赖包
激活了Pytorch之后,关掉pycharm在重新打开直接就默认进来刚创建的pytorch环境了
cd D:
cd \
3、现在验证是否安装成功,测试代码
# python版本3.9.16 import sys print(sys.version) #torch 1.7.1+cuda 11.0 import torch import torchvision print(torch.__version__) print(torch.version.cuda) print(torchvision.__version__) print(torchvision.version.cuda)
结果:
以上方式针对的是windows10操作系统,大家可以根据大家各自的操作系统选择,安装cuda或者不安装cuda
以后再使用Pycharm运行的时候不需要每次都conda activate pytorch了,直接打开pycharm用就行了。
附:
查找torch与torchvision对应版本
GitHub链接:https://github.com/pytorch/vision#installation
Index of /anaconda/cloud/pytorch/win-64/ | 北京外国语大学开源软件镜像站 | BFSU Open Source Mirror
以上均为个人经历,有参考源自水印,并不是全都是原创。
标签:torch1.7,Anaconda3,torch,pytorch,version,cuda,print,cuda11.0,安装 From: https://www.cnblogs.com/leekaihyun/p/17061911.html