pytorch1.10+cuda11.3+minconda+pycharm

pytorch1.10+cuda11.3+minconda+pycharm,第1张

Torch 1.10.1
  1. 下载并安装Cuda11.3

CUDA Toolkit 11.3 Downloads | NVIDIA Developer

  1. 下载miniconda

 

Miniconda — Conda documentation

  1. 进入conda命令行 创建虚拟环境

  

conda create -n  python=3.8 #创建虚拟环境

conda active    #激活虚拟环境

conda deactive    #退出虚拟环境
  1. 在虚拟环境中安装torch:

Previous PyTorch Versions | PyTorch

conda install pytorch==1.10.1 torchvision==0.11.2 torchaudio==0.10.1 cudatoolkit=11.3 -c pytorch -c conda-forge
  1. Pycahrm配置

(1)、添加新的解释器

 

(2)、已经存在的环境

 

(3)、找到envs下的虚拟环境pyrorch1.10下的python.exe

 

OK!

  1. 测试
import torch

print(torch.cuda.is_available())

print(torch.__version__)

print(torch.version.cuda)

print(torch.backends.cudnn.version())

 

在虚拟环境中安装jupyter notebook

conda active  #进入虚拟环境

conda install nb_conda #安装jupyter

jupyter notebook #进入

 

欢迎分享,转载请注明来源:内存溢出

原文地址:https://54852.com/langs/793955.html

(0)
打赏 微信扫一扫微信扫一扫 支付宝扫一扫支付宝扫一扫
上一篇 2022-05-06
下一篇2022-05-06

发表评论

登录后才能评论

评论列表(0条)

    保存