环境
我这里用的是Anaconda最新的版本24.9.2,创建的pyton版本是3.9,虚拟环境名叫pytorch,下载的pytorch版本是12.1
初步配置
创建虚拟环境
首先进入Anaconda的终端,然后运行conda create -n xxx python=3.9
创建一个虚拟环境,这里xxx是虚拟环境的名字,后面跟着的是python的版本
然后用conda activate xxx
激活环境,退出环境用conda deactivate xxx
如果需要删除环境,用conda rm -n xxx --all
删除
下载pytorch
然后进入pytorch的官网,下滑找到自己需要的pytroch版本,然后复制出现的命令在conda对应的虚拟环境中进行下载 (漫长的等待…这里由于网络问题会下载失败,重试几次即可,注意有显卡的不要到网上去换镜像源,新版本的pytorch用镜像源下载不成功,是CPU版本的)
下载完成后进入python环境,然后import torch
导入torch库,再用torch.cuda.isavailiable()
查看是否正确安装好pytorch,如果安装成功,有显卡的出现true,没有的出现false