题注:
这里是因为之前创建pytorch环境遇到了非常多的问题,这次重新创建一个pytorch的环境,用于以后如果需要重新安装,则根据这篇博客查考,起到一个参考博客的作用。
下载Anaconda
Anaconda下载网址:https://repo.anaconda.com/,本人测试,下载最新版本与下载之前版本都可以,这里不做纠结,直接下载最新版本。
下载Pycharm
这里下载Pycharm社区版本,最新版本。Download PyCharm: Python IDE for Professional Developers by JetBrains
利用conda创建pytorch环境:
1.在Anaconda prompt中运行以下命令:
conda create -n pytorch_new python=3.9
其中,"pytorch_new"表示创建环境的名称,"python=3.9"表示python的版本。
2.切换到创建的环境中
conda activate pytorch_new
进入如下画面:
3.查看电脑显卡支持的cuda版本:
打开系统搜索界面,搜索cmd,打开命令提示符
输入nvidia-smi,查看CUDA版本
nvidia-smi
这里的CUDA版本为12.3
4.进入pytorch官网,下载对应的pytorch
选择低于自己CUDA版本的,复制Command在创建的环境中运行,等待加载完成即可
如果没有显卡,则在Compute Platform中选择CPU版本进行下载。
当CUDA版本过低时,可选择进入英伟达官网升级显卡驱动,或是选择旧版本的Pytorch进行安装。这里选择旧版本的whl文件进行安装。
进入页面,假如CUDA版本为11.6,则搜索cu116,其中cp为python版本,选择win下的版本点击链接,下载whl文件。
最后将下载的whl文件随便存放在一个地方,然后打开cmd,输入pip install 路径\文件名.whl
,就开始离线安装了,基本上不会有什么问题。(不过首先得安装好wheel,pip install wheel
)
也可利用conda安装历史版本。进入官网。选择对应的CUDA版本,复制对应的在环境中运行即可。
检验安装成功
在虚拟环境中输入以下代码,返回True
python
import torch
torch.cuda.is_available()