首先打开
然后
创建环境指令
然后
激活对应环境
conda activate torchyo
然后进入pytorch官网
然后
然后自己找对应版本下载
复制相应指令 回车即可
安装完成后,打开pycharm
然后点开右下角 顺序点击,即可在编译器里配债好环境了。
最后打开脚本测试
import torch
print(torch.__version__) # 输出pytorch版本信息
import torchvision
print (torchvision.__version__) # 输出torchvision的版本信息
print(torch.cuda.is_available()) # 输出True 说明CUDA可用
import torch
print(torch.cuda.is_available())
print(torch.backends.cudnn.is_available())
print(torch.cuda_version)
print(torch.backends.cudnn.version())
1.7.1
0.8.2
True
True
True
11.0
8004