先安装anaconda,这里需要查看本机GPU版本,然后官网下载安装,后在anaconda环境下,安装pytorch,具体参考
在Anaconda下安装Pytorch的超详细步骤[添加链接描述](https://blog.csdn.net/qq_45281807/article/details/112442423)
小case: 本机使用的是win11,cuda11.6,安装pytorch的时候,python版本选择的是3.9
小笔记: 在anaconda prompt中设置清华园镜像
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --set show_channel_urls yes
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
如果不设置境内镜像源,会使用国外的镜像源,速度会低很多
注意: 如果切换镜像后当出现下载不了的情况,就先切换默认源,然后再修改另一个可以使用的conda源(一定要先恢复默认,再换另一个!!!)
恢复默认镜像源:
conda config --remove-key channels
注意: 在安装的时候,可能出现
Collecting package metadata (current_repodata.json): done
Solving environment: failed with initial frozen solve. Retrying with flexible solve.
Collecting package metadata (repodata.json): failed
CondaHTTPError: HTTP 000 CONNECTION FAILED for url https://repo.anaconda.com/pkgs/main/win-64/repodata.json
Elapsed: -
An HTTP error occurred when trying to retrieve this URL.
HTTP errors are often intermittent, and a simple retry will get you on your way.
If your current network has https://www.anaconda.com blocked, please file
a support request with your network engineering team.
‘https://repo.anaconda.com/pkgs/main/win-64’
大意是说:出现了网络连接错误,这个时候可以回配置文件看看,配置的几个镜像是否能正常打开,可以在安装盘目录下,搜索 .condarc ,查看内部的channels:下的地址,我的程序安装在C盘,就直接在C盘搜索 .condarc ;出来这样的文件,
当然,也可以使用 conda config --show 命令,在里面寻找channels
如果网址可以正常打开,一般情况下是临时事件,重新使用安装命令即可。
比如:此处的安装 conda的命令是 :
conda install pytorch torchvision torchaudio pytorch-cuda=11.6 -c pytorch -c nvidia
我当时重新使用安装命令三四次才成功!
检验pytorch是否安装成功: 刚才的安装界面(prompt界面),输入python,切换到python编程界面,输入 import torch,能正常导入,说明安装成功。
pycharm——file——setting——project——python interpreter——添加anaconda中的python.py