python3.5 pytorch-gpu安装记录

https://blog.csdn.net/amateur_hy/article/details/90716411

·安装pytorch 0.4.1

conda install pytorch=0.4.1 cuda90 -c pytorch

    1

以上是cuda9.0版本的安装命令。

·安装torchvision

conda install torchvision

    1

测试:

    Case-1. 确定机器的cuda是否可用、GPU的个数

     a) torch.cuda.is_available()    # check cuda是否可用
     b) nvidia-smi                   # 窗口 print 显卡信息
     c) torch.cuda.device_count()    # 统计显卡的个数
     d) torch.cuda.current_device()  # 返回当前显卡的序号[默认当前显卡index为0]
————————————————
原文链接:https://blog.csdn.net/on_theway10/article/details/90516365

 

python3.5安装一些库安不上,很多只支持3.6以上的,必须指定版本安装

pip install scikit-image==0.14(默认版本0.16换0.15不支持,降到0.14)

pip install opencv-python (找不到cv2)

pip install matplotlib==3.0 (error:Beginning with Matplotlib 3.1, Python 3.6 or above is required.)

 

(pytorch35) :~$ pip list
Package         Version    
--------------- -----------
certifi         2016.2.28  
cffi            1.10.0     
cloudpickle     1.2.2      
cycler          0.10.0     
dask            2.6.0      
decorator       4.4.0      
h5py            2.10.0     
kiwisolver      1.1.0      
llvmlite        0.30.0     
matplotlib      3.0.0      
networkx        2.4        
nibabel         2.5.1      
numba           0.46.0     
numpy           1.13.3     
olefile         0.44       
opencv-python   4.1.1.26   
Pillow          6.2.1      
pip             19.3.1     
pycparser       2.18       
pyparsing       2.4.2      
python-dateutil 2.8.0      
PyWavelets      1.1.1      
scikit-image    0.14.0     
scipy           1.3.1      
setuptools      36.4.0     
six             1.10.0     
sparse          0.8.0      
toolz           0.10.0     
torch           0.4.1.post2
torchvision     0.1.8      
tqdm            4.36.1     
wheel           0.29.0

 

你可能感兴趣的:(python3.5 pytorch-gpu安装记录)