大家好,欢迎来到IT知识分享网。
注意事项:
(1)使用anaconda安装pytorch时会自动安装cuda和cudnn
(2)anaconda会自动安装python和一些基础包
所以无需单独安装python、cuda、cudnn。
配置过程:
1安装conda。官网下载anaconda,直接安装即可,系统环境中手动添加环境变量
2更换清华源。打开 cmd(win+R,输入 cmd 后回车),输入 conda info,如果正确无误的话,会显示 conda 的信息。为了 conda 的下载速度,需要添加国内的清华源,输入:conda config –set show_channel_urls yes
创建配置文件 .condarc,该文件位于:C:\Users\你的用户名 下。用文本编辑器打开它,然后复制以下内容完全覆盖已有内容:
ssl_verify: true
show_channel_urls: true
auto_activate_base: false
channels:
- http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/win-64/
- http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/win-64/
- https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
- https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
- https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
3创建新环境。打开anaconda prompt,新建一个名为pytorch的conda环境
conda create -n pytorch python=3.7(python版本可以自己设置)
4打开NVIDIA控制面板,查看驱动版本,打开pytorch官网,根据版本会得到一个安装命令,类似这种:
conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch
5在anaconda prompt 激活pytorch环境:activate pytorch
粘贴pytorch官网得到的安装命令,把-c pytorch去掉(-c pytorch默认pytorch从官网下载,速度很慢),网速好的话应该很快。注意有没有报错,报错的话重新执行一遍安装命令。
6 pycharm官网下载社区版,一路安装。新建python文件,然后打开
Settings -> Project Interpreter, 这时会发现项目解释器是PyCharm自带的或者是Anaconda自带的python.exe,替换成我们安装的pytorch环境的python解释器即可。
在.py文件中输入
import torch
torch.cuda.is_available()
运行显示true,表示torch环境配置成功且cuda和cudnn可以使用,安装完毕。
参考资料:
安装博客1
cuda和cudnn问题
清华源
免责声明:本站所有文章内容,图片,视频等均是来源于用户投稿和互联网及文摘转载整编而成,不代表本站观点,不承担相关法律责任。其著作权各归其原作者或其出版社所有。如发现本站有涉嫌抄袭侵权/违法违规的内容,侵犯到您的权益,请在线联系站长,一经查实,本站将立刻删除。 本文来自网络,若有侵权,请联系删除,如若转载,请注明出处:https://yundeesoft.com/11731.html