Python3.9+torch1.7.1+cuda11.0+cudnn8.0+Anaconda3安装

前言
想要安装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



 以上均为个人经历,有参考源自水印,并不是全都是原创。

posted @ 2023-01-19 18:26  LeeKaiHyun  阅读(954)  评论(0编辑  收藏  举报