首先查看当前的torch版本
激活conda环境,并查看
conda list
发现pytorch版本是cpu_py38*********,代表是CPU版本。
解决方法-重装(指定安装pytorch-gpu版本)
(别太在意原有的库包,因为cpu版本关联了很多包,一个是cpu版本会有不少包都是cpu版本,重装后缺什么再安装什么就是了)
重装环境
我这里指定3.8的py版本重装一个叫env的环境
conda create -n env python=3.8
查看当前的conda源中,pytorch的安装包信息
conda search pytorch
输出:
!!!重点!!!!!!!!!!!!
直接安装pytorch的gpu版本,注意对应你当前conda虚拟环境的python的版本,我这里是py38,于是:
conda install pytorch=1.12.1=gpu_cuda113py38h19ae3d8_1
此方法可以连带安装cuda和cudnn:
另外,如果速度很慢,就是当前的conda源问题,这里提供我的:
channels:
- defaults
show_channel_urls: true
default_channels:
- http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
- http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free
- http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/r
custom_channels:
conda-forge: http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
msys2: http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
bioconda: http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
menpo: http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
pytorch: http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
simpleitk: http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
不明白的可以留言,看到一定回复。
文章出处登录后可见!
已经登录?立即刷新